import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;

public class FindRepeatedDnaSequences {

    @Test
    void test1(){
        List<String> result = findRepeatedDnaSequences("AAAAACCCCCAAAAACCCCCCAAAAAGGGTTT");
        HashSet<String> set = new HashSet<>(result);
        assertTrue(set.contains("AAAAACCCCC"));


        assertTrue(set.contains("CCCCCAAAAA"));

        assertEquals(2,result.size());
    }

    @Test
    void test2(){
        List<String> res = findRepeatedDnaSequences("AAAAAAAAAAAAA");
        assertEquals("AAAAAAAAAA",res.get(0));
        assertEquals(1,res.size());
    }

    @Test
    void test3(){
        List<String> result = findRepeatedDnaSequences("AAAAAAAAAAA");
        assertTrue(result.contains("AAAAAAAAAA"));
        assertEquals(1,result.size());
    }

    /**
     * 187 重复的DNA序列
     * 在研究 DNA 时，识别 DNA 中的重复序列非常有用。
     *
     * 给定一个表示 DNA序列 的字符串 s ，返回所有在 DNA 分子中出现不止一次的 长度为 10 的序列(子字符串)。
     * 你可以按 任意顺序 返回答案。
     * @param s
     * @return
     */
    public List<String> findRepeatedDnaSequences(String s) {

        HashMap<String, Integer> map = new HashMap<>();

        HashSet<String> result = new HashSet<>();

        int length = s.length();

        for (int i = 10; i < length; i++) {

            map.put(s.substring(i-10,i),i-10);

            String key = s.substring(i-9, i + 1);

            Integer index = map.get(key);

            if(index!=null){
                result.add(key);
            }
        }

        return new ArrayList<>(result);
    }
}
